#!/bin/bash

# 统计当前目录下所有.py, .yaml, .sh文件所占总大小以及单独列出每类文件的大小

echo "各类型文件占用空间统计："
echo "--------------------------"

echo -n ".py  文件总大小:   "
find . -type f -name "*.py" -print0 | du --files0-from=- -ch | grep total$ || echo "0"

echo -n ".yaml文件总大小:   "
find . -type f -name "*.yaml" -print0 | du --files0-from=- -ch | grep total$ || echo "0"

echo -n ".sh  文件总大小:   "
find . -type f -name "*.sh" -print0 | du --files0-from=- -ch | grep total$ || echo "0"

echo "--------------------------"
echo -n "三类文件合计大小:  "
find . -type f \( -name "*.py" -o -name "*.yaml" -o -name "*.sh" \) -print0 | du --files0-from=- -ch | grep total$ || echo "0"

echo
echo "各子文件夹占用空间统计："
echo "--------------------------"
# 列出所有一级子目录（不包括.本目录），并统计大小
find . -mindepth 1 -maxdepth 1 -type d | while read dir; do
    du -sh "$dir"
done
